python - boolean mask on array of custom objects